def show_all_data_augmentation(video_id, frame_id):
# Original Image with Bounding Box
original_image_bb = show_COT_img_BB(video_id, frame_id)
# Apply Hue, Saturation, and Value transformation
transformed_img_hue_sat, transformation_title_hue_sat = apply_hue_saturation_value_transform(
video_id, frame_id,
hue_shift_limit=(-20, 20),
sat_shift_limit=(-30, 30),
val_shift_limit=(-20, 20),
p=1
)
# Apply Uniform Box Blur
transformed_img_uniform_box_blur, transformation_title_uniform_box_blur = apply_uniform_box_blur(
video_id, frame_id,
blur_limit=(9, 15),
p=1
)
# Apply Sun Flare effect
transformed_img_sun_flare, transformation_title_sun_flare = apply_sun_flare(
video_id, frame_id,
flare_roi=(0.1, 0, 0.9, 0.3),
angle_range=(0.15, 0.45),
num_flare_circles_range=(5, 15),
src_radius=200,
src_color=(255, 200, 100),
method="physics_based",
p=1.0
)
# Apply Zoom Blur
transformed_img_zoom_blur, transformation_title_zoom_blur = apply_zoom_blur(
video_id, frame_id,
max_factor=(1, 1.10),
step_factor=(0.01, 0.011),
p=1
)
# Apply Random Brightness and Contrast
transformed_img_random_brightness_contrast, transformation_title_random_brightness_contrast = apply_random_brightness_contrast(
video_id, frame_id,
brightness_limit=(-0.2, 0.2),
contrast_limit=(-0.3, 0.3),
p=1
)
# Apply Horizontal Flip
transformed_img_horizontal_flip, transformation_title_horizontal_flip = apply_horizontal_flip(
video_id, frame_id,
p=1.0
)
# Apply Vertical Flip
transformed_img_vertical_flip, transformation_title_vertical_flip = apply_vertical_flip(
video_id, frame_id,
p=1.0
)
# Apply Rotation Transform
transformed_img_rotation, transformation_title_rotation = apply_rotation_transform(
video_id, frame_id,
value=19,
limit=(-50, -30),
interpolation=cv2.INTER_LINEAR,
border_mode=cv2.BORDER_REFLECT_101,
p=1
)
# Apply Random Affine Transform
transformed_img_random_affine, transformation_title_random_affine = apply_random_affine_transform(
video_id, frame_id,
shift_limit=(-0.2, 0.3),
scale_limit=(0.8, 0.9),
rotate_limit=(20, 45),
interpolation=cv2.INTER_LINEAR,
p=1
)
# Apply Coarse Dropout
transformed_img_coarse_dropout, transformation_title_coarse_dropout = apply_coarse_dropout(
video_id, frame_id,
num_holes_range=(8, 15),
hole_height_range=(0.02, 0.05),
hole_width_range=(0.02, 0.05),
fill_value="random",
p=1.0
)
# Apply Pixel Dropout
transformed_img_pixel_dropout, transformation_title_pixel_dropout = apply_pixel_dropout(
video_id, frame_id,
dropout_prob=0.1,
per_channel=False,
drop_value=0,
mask_drop_value=3,
p=1.0
)
# List of Transformed Images and Titles
transformed_images = [
original_image_bb,
transformed_img_hue_sat,
transformed_img_uniform_box_blur,
transformed_img_sun_flare,
transformed_img_zoom_blur,
transformed_img_random_brightness_contrast,
transformed_img_horizontal_flip,
transformed_img_vertical_flip,
transformed_img_rotation,
transformed_img_coarse_dropout,
transformed_img_pixel_dropout,
transformed_img_random_affine
]
transformation_titles = [
f"Original Image | Video {video_id} Frame {frame_id}",
transformation_title_hue_sat,
transformation_title_uniform_box_blur,
transformation_title_sun_flare,
transformation_title_zoom_blur,
transformation_title_random_brightness_contrast,
transformation_title_horizontal_flip,
transformation_title_vertical_flip,
transformation_title_rotation,
transformation_title_coarse_dropout,
transformation_title_pixel_dropout,
transformation_title_random_affine
]
plt.figure(figsize=(14, 12))
for idx, (img, title) in enumerate(zip(transformed_images, transformation_titles)):
plt.subplot(4, 3, idx + 1)
plt.imshow(img)
plt.title(title, fontsize=10)
plt.axis('off')
plt.tight_layout()
plt.show()
# Cleanup inside the function
del (
original_image_bb,
transformed_img_hue_sat, transformed_img_uniform_box_blur, transformed_img_sun_flare,
transformed_img_zoom_blur, transformed_img_random_brightness_contrast,
transformed_img_horizontal_flip, transformed_img_vertical_flip,
transformed_img_rotation, transformed_img_random_affine,
transformed_img_coarse_dropout, transformed_img_pixel_dropout,
transformed_images, transformation_titles
)
# gc.collect()
